#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
AI Agent 快速开始示例
演示如何使用 AI Agent 执行简单的任务
"""
from colorama import init, Fore, Style

from agent import AIAgent

# 初始化colorama
init()

def interactive_demo():
    """交互式演示"""
    print(f"{Fore.CYAN} AI Agent 交互式演示{Style.RESET_ALL}")
    print("输入 'exit' 退出")
    print("-" * 30)

    agent = AIAgent("openai")

    while True:
        try:
            task = input(f"{Fore.GREEN} 请输入任务 > {Style.RESET_ALL}").strip()

            if task.lower() in ['exit', 'quit']:
                print(f"{Fore.YELLOW}再见！{Style.RESET_ALL}")
                break
            if not task:
                continue
            print()
            result = agent.execute_task(task)
            print()
            print(f"{Fore.MAGENTA} 任务总结：{Style.RESET_ALL}")
            print(result["summary"])
            print()
        except KeyboardInterrupt:
            print(f"\n{Fore.YELLOW} 输入 'exit' 退出{Style.RESET_ALL}")
        except Exception as e:
            print(f"{Fore.RED} 错误：{str(e)}{Style.RESET_ALL}")

if __name__ == "__main__":
    interactive_demo()


